package com.ruifu.conversions.utils;

import com.aspose.cells.License;
import com.aspose.cells.PdfSaveOptions;
import com.aspose.cells.Workbook;
import lombok.extern.slf4j.Slf4j;

import java.io.*;

/**
 * @author lcq
 * @create 2020/5/22
 */
@Slf4j
public class ExcelToPdfUtil {


    /**
     * 获取license 去除水印
     * @return
     */
    public static boolean getLicense() {
        boolean result = false;
        try {
            InputStream is = ExcelToPdfUtil.class.getClassLoader().getResourceAsStream("\\license.xml");
            License aposeLic = new License();
            aposeLic.setLicense(is);
            result = true;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return result;
    }

    public static void excel2pdf(String sourceFilePath, String desFilePathd ){
        // 验证License 若不验证则转化出的pdf文档会有水印产生
        if (!getLicense()) {
            return;
        }
        try {
            // 原始excel路径
            Workbook wb = new Workbook(sourceFilePath);
            FileOutputStream fileOS = new FileOutputStream(desFilePathd);
            PdfSaveOptions pdfSaveOptions = new PdfSaveOptions();
            //缩放到一个页面（如果列太多 太长）
            pdfSaveOptions.setOnePagePerSheet(true);
            //** 自动缩放了
            int[] autoDrawSheets={3};
            //当excel中对应的sheet页宽度太大时，在PDF中会拆断并分页。此处等比缩放。
            autoDraw(wb,autoDrawSheets);
            //以下代码打开则只显示第一个sheet
            int[] showSheets={0};
            //隐藏workbook中不需要的sheet页。
             printSheetPage(wb,showSheets);
            wb.save(fileOS, pdfSaveOptions);
            fileOS.flush();
            fileOS.close();

        } catch (Exception e) {
            throw new RuntimeException();
        }
    }

    /**
     * excel转pdf
     * @param inbytes
     * @return
     * @throws IOException
     */
    public static byte[] excel2pdf(byte[] inbytes) throws IOException {
        // 验证License 若不验证则转化出的pdf文档会有水印产生
        if (!getLicense()) {
            return null;
        }
        ByteArrayOutputStream pdfstream=null;
        try {
            // 原始excel流
            Workbook wb = new Workbook(new ByteArrayInputStream(inbytes));
            //保存转成pdf的流
            pdfstream = new ByteArrayOutputStream();
            PdfSaveOptions pdfSaveOptions = new PdfSaveOptions();
            //缩放到一个页面（如果列太多 太长）
            pdfSaveOptions.setOnePagePerSheet(true);
            wb.save(pdfstream, pdfSaveOptions);
            byte[] outbytes = pdfstream.toByteArray();
            return outbytes;
        } catch (Exception e) {
            log.error(""+e);
            return null;
        }finally {
            if (pdfstream!=null){
                pdfstream.close();
            }
        }
    }


    /**
     * @param wb 设置打印的sheet自动拉伸比例
     * @param page 自动拉伸的页的sheet数组
     */
    public static void autoDraw(Workbook wb, int[] page){
        if (null!=page&&page.length>0){
            for (int i = 0; i < page.length; i++) {
                wb.getWorksheets().get(i).getHorizontalPageBreaks().clear();
                wb.getWorksheets().get(i).getVerticalPageBreaks().clear();
            }
        }
    }


    /**
     * 隐藏workbook中不需要的sheet页。
     * @param wb
     * @param page 显示页的sheet数组
     */
    public static void printSheetPage(Workbook wb, int[] page){
        for (int i= 1; i < wb.getWorksheets().getCount(); i++)  {
            wb.getWorksheets().get(i).setVisible(false);
        }
        if (null==page||page.length==0){
            wb.getWorksheets().get(0).setVisible(true);
        }else{
            for (int i = 0; i < page.length; i++) {
                wb.getWorksheets().get(i).setVisible(true);
            }
        }
    }
}
